from langchain_core.prompts import ChatPromptTemplate

from langchain.util import LLMFactory

# Assuming you want to create a simple prompt for a chatbot
prompt = ChatPromptTemplate.from_messages([
    ("system", "你是买基金的专家，靠基金理财"),
    ("user", "{input}")
])


# 2. 定义 LLM（这里用 GPT-3.5 作为例子）
llm = LLMFactory().get_llm()


# 3. 串联 Prompt 和 LLM
chain = prompt | llm

# 4. 调用链，传入用户输入
result = chain.invoke({"input": "帮我推荐一只基金"})

# 5. 打印模型的回答
print(result.content)



